package classificacao;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Vector;

import InterfacePersistencia.XMLXStream.ReadXMLXStream;
import InterfacePersistencia.XMLXStream.WriteXMLXStream;
import InterfacePersistencia.conexao.Messages;

public class TesteClassificacao {
	
private Messages msg;
	
	public void teste (){
		WriteXMLXStream xml;
		HashMap<String, Integer> wordSPAM = new HashMap<String, Integer>();
		HashMap<String, Integer> wordLEGIT = new HashMap<String, Integer>();
				
		msg = new Messages();
		
	/*	// MAP de palavras Spam;
		wordSPAM.put("rafael", 9);
		wordSPAM.put("joao", 2);
		wordSPAM.put("Maria", 1);
		
		// MAP palavras legitimas
		wordLEGIT.put("joaoo", 1);
		wordLEGIT.put("rafael", 1);
    
		Estatisticas est = new Estatisticas();

		// set Probabilidade maior chance de ser span
		est.setPCL(0.5);
		est.setPCS(0.5);

		// set Map de spam e Legitimos
		est.setWordsCS(wordSPAM);
		est.setWordsCL(wordLEGIT);
		est.setNdocsTreinados(10);
		est.setNdocsCL(2);
		est.setNdocsCS(50);

	//	xml = new WriteXMLXStream(est);

		Estatisticas esta = new Estatisticas();

	//	esta = (Estatisticas) ReadXMLXStream.ReadXML(esta);

		System.out.println("O valor da probabilidade he:" + esta.getPCL());

		// Vector: mensagem simulada'
		
		*/
		Vector<Word> vec = new Vector<Word>();
		ArrayList<Word> arr = new ArrayList<Word>();

		vec.add(new Word("rafael"));
		vec.add(new Word("joao"));

		arr.add(new Word("emerson"));
		arr.add(new Word("nada"));

		
	
		NBayes nb = new NBayes();
		msg.addPalavras("Maria");
		msg.addPalavras("joao");
		msg.addPalavras("emerson");
		nb.treino(msg, false);
		
		//nb.getArrayWord();
		
		
		nb.classify(arr);

	}
	public static void main(String[] arg) {
		TesteClassificacao ts = new TesteClassificacao();
		ts.teste();
	}	
}